added exercise 2 bonus