some improvements in ML_control