@Nullable
    private static KtClassBody getClassBody(PsiElement element) {
        if (element instanceof KtClassOrObject) {
            return ((KtClassOrObject) element).getBody();
        } else {
            return null;
        }
    }
@Nullable
    private static KtClassBody getClassBody(PsiElement element)