Added part 3 of assignment