Fix streamlit-folium incompatibility (add layer to map with new class)

Fix streamlit-folium incompatibility (add layer to map with new class) #589