diff --git a/main.go b/main.go
index b48624f..8c78ffe 100644
--- a/main.go
+++ b/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"time"
 	"log"
+    "flag"
 	"fmt"
 	"database/sql"
 	_ "github.com/go-sql-driver/mysql"