annotate flow for code gen