changes made to test.py