Add training state