Finished part A of Ass2/Prob2