Added removeClass functionality