Added additional tests to the flow.