Use data attribute instead of class name