Simple trees flow script for stage