Fix addClass