Make some small changes to make fit function converges faster. Stuck at the shared covariance.