Update multi_step_cov_estimator.py