Testing sgd
Started on NIPs midway