Add support to customize the map