trying to finish hw2