Update HW_2extracredit.py

fix filename for output