Done with GMM EM and added some nice plots