Added models.py with ModelBase