Fresh re-write of code up till sgd iter
Store
merge