Added save method for models