Add a model saving test