Clean up learn.py, pickle model.