test.py updated