add needed class