working update for training