Add positioning of flow stage