Add location-aware coloring to prompt