Make homework 1 task 2 more coherent