improve example for exercise 2