Pickle final model
Start on prediction code