Add model save and load functions
Add model save and load funcitons