adding models.py