Added some more to the models.py