change remove class function