Added integration tests for models.py