Removed redundant code from train_validate_test