modified 0x01.py