Update models.py

Adding classes to take advantage of inheritance