Added save method to model