Update gmm.py