Add flow control statements