Added some helper functions to save and load the logit models