Working on test0901.py