Add util functions to add, remove and check for classes in elements