diff --git a/main.lua b/main.lua
index e309e5f..379419c 100644
--- a/main.lua
+++ b/main.lua
@@ -22,6 +22,7 @@ local settings = {
 	depth = 7,
 	
 	seed = os.time(),
+	-- TODO: Add branching factor setting here & associated CLI argument
 	filename = "maze.scad",
 	
 	scale = 5,
@@ -96,6 +97,9 @@ for i=1,#arg do
 	elseif arg[i] == "--filename" or arg[i] == "-f" then
 		i = i + 1
 		settings.filename = arg[i]
+	elseif arg[i] == "--colour" then
+		i = i + 1
+		settings.colour = arg[i]
 	elseif arg[i] == "--scale" or arg[i] == "-m" then
 		i = i + 1
 		settings.scale = tonumber(arg[i])