updated to 2D GMM parameter estimation