changes to test.py