Assignment part 3 done