modification for training