Update on test.py