Got the flow working end to end