private static PyStatementList getStatementList(final @NotNull PsiElement elementToMove) {
    return PsiTreeUtil.getParentOfType(elementToMove, PyStatementList.class, true,
                                                                PyStatementWithElse.class, PyLoopStatement.class,
                                                                PyFunction.class, PyClass.class);
  }
private static PyStatementList getStatementList(final @NotNull PsiElement elementToMove)