speed up and fix to sgd