made changes for models