Update of test.py