setting up the exercise 2 solution too