made changes to the models