Add test of control flow