Merge SGD MSE changes into master