Add legend functionality for colored locations