Create HW3code.py