put the classes in FlowTestBuilder in their own file