Left some commented out code in TrainTests