Changed selector to class