slight change to models file