HW1 first attempt