Add helper functions
Disable less useful plot
Start fitting the regression dataset