Create hw_6_draft.py