Change selector from class to data attribute