Done with part B! But not quite sure how to test it yet