Modified test.py.