@NotNull
  public static List<GrStatement> collectReturns(@Nullable GroovyPsiElement element) {
    return collectReturns(element, element instanceof GrCodeBlock || element instanceof GroovyFile || element instanceof GrLambdaBody);
  }
@NotNull
  public static List<GrStatement> collectReturns(@Nullable GroovyPsiElement element)