Added functions for add and remove classes for elements