add hasClass check