Add removeClass method