Integrate code, update for better regressor (Roughly version, haven't update the new changed plot for MOMLO yet).