Added models.py and tests.py