add class in directive