Working on exercise 4.4