Now it can parse while/if/switch statements