Assignment 2: Python Worm Program Solution