Updates  for new model