updated the flags used

1 files changed, 6 insertions, 2 deletions

diff --git a/src/flags.go b/src/flags.go
index 1fd28df..df7ee52 100644
--- a/src/flags.go
+++ b/src/flags.go
@@ -5,11 +5,15 @@ import "flag"
 func parseFlags() config {
 	sshPort := flag.Int("sshPort", 2222, "the port the ssh server is listening on")
 	httpPort := flag.Int("httpPort", 8084, "the port the http server is listening on")
+	bindIP := flag.String("bind", "0.0.0.0", "the ip the servers should be listening on")
+	stateFile := flag.String("stateFile", "./statefile.txt", "the file path to store the state in")
 
 	flag.Parse()
 
 	return config{
-		sshPort:  *sshPort,
-		httpPort: *httpPort,
+		sshPort:   *sshPort,
+		httpPort:  *httpPort,
+		bindIP:    *bindIP,
+		stateFile: *stateFile,
 	}
 }