made changes to test.py