slight model changes