add some changes to the regression demo