Add test for for/switch