1st attempt at python portion of HW1