Added basic testing for control statements