More sophisticated training program
Better hyperparameters