switch statement in python 3