change js to add/remove class