add addClass and removeClass functions