Write a code in Python to find the area of a circle using its radius.
import math

def circle_area(radius):
  """
  Returns the area of a circle based on its radius
  """
  return math.pi * (radius**2)

if __name__ == '__main__':
    print(circle_area(4))