removed bug and added plotting of model updates