Changed it to filter by either class or data-