Switched NN to MSE for regression