Added models.py with all related stuff