changed the structure of the models.py file