Added example code to models.py