Add addClass and removeClass