Add in polygon functionality