Cleaned up TrainNN2 a bit.