sum the area and square of the perimeter of the regular polygon

from math import *
        '''
        A regular polygon has n number of sides. Each side has length s.
        The perimeter of a polygon is: length of the boundary of the polygon
        Write a function called polysum that takes 2 arguments, n and s. 
        This function should sum the area and square of the perimeter of 
        the regular polygon. 
        The function returns the sum, rounded to 4 decimal places.
        '''
def polysum (n, s):
    per = n * s
    area = (0.25 * n * (s**2))/tan(pi/n)
    return round((area + per**2), 4)

print(polysum(31, 32))