Update sgd_dictionary.py