added second line to test.py