Adding test_models.py