Updated choropleth example to use the new API