Can call functions and print text.

1 files changed, 1 insertions, 2 deletions

diff --git a/parser.py b/parser.py
index e3da802..b376dbf 100755
--- a/parser.py
+++ b/parser.py
@@ -150,8 +150,7 @@ def test_parser():
     print(build_ast(code))
 
 def build_ast(code):
-    if not code.endswith('\n'):
-        code = code + '\n'
+    code = code.rstrip() + '\n'
     lex.lex()
     parser = yacc.yacc()
     result = parser.parse(code)