Add logic for coloring a div on click