HW1 python code added