Figured out exercise 5b