some changes to test.py