small changes to model