fixes from the training part #2