private static void applyChanges(@NotNull Mover mover,
                                   @NotNull ReturnContext context,
                                   boolean removeReturn,
                                   @NotNull ModPsiUpdater updater) {
    for (PsiElement anchor : mover.insertBefore) {
      PsiElement added = anchor.getParent().addBefore(context.returnStatement, anchor);
      updater.highlight(added);
    }
    mover.replaceInline.forEach(e -> {
      if (e instanceof PsiBreakStatement breakStatement) {
        replaceStatementKeepComments(breakStatement, context.returnStatement, updater);
      }
      else if (e instanceof PsiAssignmentExpression assignment) {
        inlineAssignment(assignment, context.returnStatement, updater);
      }
    });
    mover.removeCompletely.forEach(e -> removeElementKeepComments(e));
    if (removeReturn) {
      removeReturn(context);
    }
  }
private static void applyChanges(@NotNull Mover mover,
                                   @NotNull ReturnContext context,
                                   boolean removeReturn,
                                   @NotNull ModPsiUpdater updater)