changed add/remove class to attr class
changed attr class to className