Adding contexts to the flowchart