Add control structures tests