@Override
  public void beforeMove(@NotNull Editor editor, @NotNull MoveInfo info, boolean down) {
    super.beforeMove(editor, info, down);

    PsiElement statement = STATEMENT_TO_SURROUND_WITH_CODE_BLOCK_KEY.get(info);
    if (statement != null) {
      surroundWithCodeBlock(info, down, statement);
    }
  }
@Override
  public void beforeMove(@NotNull Editor editor, @NotNull MoveInfo info, boolean down)