Add argument 'map_type' to function