small changes to flowchart