add class to parent element instead #3