change if then set  class