Edit an existing Python program to calculate the area of a circle.
import math
def area(radius):
    return math.pi * radius ** 2
import math
def area(radius):
    area = math.pi * radius ** 2
    return area