Added test case for control flow constructs