private void parseReturnStatement(SyntaxTreeBuilder builder) {
    LOG.assertTrue(builder.getTokenType() == PyTokenTypes.RETURN_KEYWORD);
    final SyntaxTreeBuilder.Marker returnStatement = builder.mark();
    builder.advanceLexer();
    if (builder.getTokenType() != null && !getEndOfStatementsTokens().contains(builder.getTokenType())) {
      getExpressionParser().parseExpression();
    }
    checkEndOfStatement();
    returnStatement.done(PyElementTypes.RETURN_STATEMENT);
  }
private void parseReturnStatement(SyntaxTreeBuilder builder)