solve part 1 and 2 of assignment 6