Removed the check for the training dir; mkdir does nothing if the directories already exist, so it is always safe to run the command.