Second try on HW1 after doing HW2