Added covariance_type option to fit