updates to code to use folium