modify tests.py and models.py