adjusted test.py