finished updating test.py