Escribir un programa que calcule el área superficial de un cono recto

from math import pi
def main():
  r = input("Enter radius: ")
  h = input("Enter height: ")
  area = pi * r * (r + sqrt(h*h + r*r))
  print area