Add switch statement to the AST model