1 files changed, 17 insertions, 9 deletions
diff --git a/src/Statements/CreateStatement.php b/src/Statements/CreateStatement.php
index edb87ea..69e3f7f 100644
--- a/src/Statements/CreateStatement.php
+++ b/src/Statements/CreateStatement.php
@@ -57,20 +57,14 @@ class CreateStatement extends Statement
     public $options;
 
     /**
-     * The parameters of this routine.
-     *
-     * @var ParamDefFragment[]
-     */
-    public $parameters;
-
-    /**
-     * The options of the table.
+     * The options of the entity (table, procedure, function, etc.).
      *
      * @var OptionsFragment
      *
      * @see CreateDefFragment::$TABLE_OPTIONS
+     * @see CreateDefFragment::$FUNC_OPTIONS
      */
-    public $tableOptions;
+    public $entityOptions;
 
     /**
      * Field created by this statement.
@@ -80,6 +74,20 @@ class CreateStatement extends Statement
     public $fields;
 
     /**
+     * The return data type of this routine.
+     *
+     * @var DataTypeFragment
+     */
+    public $return;
+
+    /**
+     * The parameters of this routine.
+     *
+     * @var ParamDefFragment[]
+     */
+    public $parameters;
+
+    /**
      * The body of this function or procedure.
      *
      * @var Token[]