minor updates to model