HW1 - Written in Python