added addClass and removeClass