@Override
  protected @Nullable Presentation getPresentation(@NotNull ActionContext context, @NotNull PsiElement element) {
    return findMethod(element, context.offset()) != null ? 
           Presentation.of(GroovyBundle.message("add.return.type")) : null;
  }
@Override
  protected @Nullable Presentation getPresentation(@NotNull ActionContext context, @NotNull PsiElement element)