HW 1 Python Script - Finished