Add / remove classes on click on graph