@Override
  public boolean isEnabledOnElements(PsiElement[] elements) {
    return elements.length == 1 && elements[0] instanceof PyClass;
  }
@Override
  public boolean isEnabledOnElements(PsiElement[] elements)