Changed learning to stochastic gradient descent and removed validation set