Bug fix

On save functions the current model is updated instead of being created fresh, so its able to be used after save.

i.e. the passed by referenced model is returned instead of a new model.