Switch in Python