Add ability to pickle fit output