adding models.py
added models
I fucked up