Update models.py

First pass at models