@Nullable
  public PsiElement getHighlightingElement(@NotNull GrExpression expression) {
    if (expression instanceof GrUnaryExpression unary) {
      return unary.getOperationToken();
    }

    if (expression instanceof GrBinaryExpression binary) {
      return binary.getOperationToken();
    }

    if (expression instanceof GrSafeCastExpression safeCast) {
      return safeCast.getOperationToken();
    }
    return null;
  }
@Nullable
  public PsiElement getHighlightingElement(@NotNull GrExpression expression)