fixed a bug that occured when output layer wasnt softmax. Added mse to printout