Highlight Filter elements

Hi there, it will be great if we can somehow differentiate differnt kind of filters selected with some light color difference ? currently if i select , say Heading , Image and Text, i am lost . i need to scroll through one by one to check it. Idea would be if you can apply same color for filter selection and the element itselves. example Heading filter, heading element. = light gray. Image filter, Image element. = light green . etc cheers