diff --git a/src/main/wacc/parser.scala b/src/main/wacc/parser.scala
index 71198b2..6a3407d 100644
--- a/src/main/wacc/parser.scala
+++ b/src/main/wacc/parser.scala
@@ -121,12 +121,17 @@ object parser {
 
   // Statements
   private lazy val `<program>` = Program(
-    "begin" ~> many(
-      atomic(`<type>`.label("function declaration") <~> `<ident>` <~ "(") <**> `<partial-func-decl>`
-    ).label("function declaration"),
-    (atomic(`<ident>` <~ "(").verifiedExplain("function is missing return type") | `<stmt>`.label(
+    "begin" ~> (
+      many(
+        atomic(
+          `<type>`.label("function declaration") <~> `<ident>` <~ "("
+        ) <**> `<partial-func-decl>`
+      ).label("function declaration") |
+        atomic(`<ident>` <~ "(").verifiedExplain("function declaration is missing return type")
+    ),
+    `<stmt>`.label(
       "main program body"
-    )) <~ "end"
+    ) <~ "end"
   )
   private lazy val `<partial-func-decl>` =
     FuncDecl(