updated to class selector to reflect latest changes