what about using ngClass instead here?