add models.py file