Updated test01.py