Add map handling