LaTeX templating and bills begin

1 files changed, 7 insertions, 1 deletions

diff --git a/main.go b/main.go
index 32861c0..6cc9185 100644
--- a/main.go
+++ b/main.go
@@ -14,7 +14,7 @@ import (
 
 //var svar string
 var starttime,stoptime string
-var projectid, edittaskid, editprojectid int
+var projectid, edittaskid, editprojectid, editcustomerid int
 var deltask int
 var addcustomer bool
 var newproject, newtask, stoptask, allproj, runinter, test bool
@@ -65,6 +65,10 @@ func init() {
         "editproject",
         0,
         "Specify which project to edit")
+    flag.IntVar(&editcustomerid,
+        "editcustomer",
+        0,
+        "Specify which customer to edit")
     flag.IntVar(&projectid,
         "project",
         0,
@@ -183,6 +187,8 @@ func main() {
         editTask(edittaskid)
     } else if (editprojectid>0) {
         editProject(editprojectid)
+    } else if (editcustomerid>0) {
+        editCustomer(editcustomerid)
     } else if (projectid>0) {
         setProject(projectid)
         stdOut()