Added some tests for models.py