Almost done with exercise 4