Create models.py

Models module