Add helper function to get switch cases