Remove add class method