cleaned test.py