Exercise 2 - extension to exercise 1
Exercise 2 - extension to exercise 1