Updating context to new model