def f(n):
    return n * 2 if <selection>n</selection> else n + 1