HW0_part3 cleaning 2