Added state polygons and made vis improvements