Add conditional for map display