Solution for graded exercise 2