@Override
  public boolean isAccepted(@NotNull PsiElement element) {
    return element instanceof PyClass || element instanceof PyFunction;
  }
@Override
  public boolean isAccepted(@NotNull PsiElement element)