Minor update

Minor changes to models.py