changes to test_models.py