New version of test.py