Add test to remove and toggle class.