Fixed a couple of bugs in the regression demo notebook