small fixes
actually reverting sgd changes