Adding test for control statements