Changed this to part 2 of the exercise