Working on exercise 2 and 3