made exercise 4 more interesting