modified apo_gmm.py