Created to check whether the training works