Model now updates completely with PDE changes