Add support for branch in switch expression