Almost finished with Part 1 HW2