Version 2 of test.py