finally cracked challenge4!
ex 5