Adding control flow