Prosty program w Python'ie

Chciałbym wiedzieć jak poprawić poniższy kod, żeby działał, jak w poleceniu

napisz program który zwróci x+1 dla x<0, x dla x=<0,1>, x2-1 dla x>1)

def licz():
      if x <0:
            return x +1
      elif x==0:
            return x
      else:
           return  (x**2 +1)
            

def licz(x):
    if x < 0:
        return x + 1
    elif x >= 0 and x <= 1:
        return x
    else:
        return (x ** 2 + 1)

def licz(x):
    return (x + 1) if x < 0 else x if x<=1 else (x**2 + 1)

https://ideone.com/lp6jK6