import tree
def main():
   msg =tree.logic(data)
   print msg

def logic(data):
    if data["A"] > data["B"]:
        return "A is bigger"
    else:
        return "B is bigger"

def logic(data):
    return "Is it returning?"