Create hw2_draft.py