more changes to model