Updates to code flow sample