important small changes to model