Add control flow