HW2
Some parts of the GMM are working