Annotating some functions with flow