Better model params
Plotting lines
Setting last_prediction in predict as well
Resetting last prediction in predictor.resetState
Better logging in main