@Override
  public boolean isEnabledOnElement(PsiElement element) {
    return element instanceof PsiMethod || element instanceof PsiClass;
  }
@Override
  public boolean isEnabledOnElement(PsiElement element)