minor changes to model