update for model changes