adding test units + improving the comments in sgd.py