add/remove class fix