Create models.py

wip