Add switch statement visualization