Adding save method, to persist state of model