take a number and respond with a corresponding poem

def poem(num):
  if num == 1:
    return "One fish, two fish"
  elif num == 2:
    return "red fish, blue fish"
  else:
    return "what's the haps?"
    
poem(1)