add corrections to hw2