Add solution to exercise from Part 4