3 files changed, 16 insertions, 9 deletions

diff --git a/bridge.go b/bridge.go
index ea6d2ce..90f7719 100644
--- a/bridge.go
+++ b/bridge.go
@@ -1,4 +1,4 @@
-package main
+package hue
 
 // username: 319b36233bd2328f3e40731b23479207
 import (
@@ -11,13 +11,6 @@ import (
     "fmt"
 )
 
-func main() {
-    bridge := NewBridge("192.168.1.128", "319b36233bd2328f3e40731b23479207")
-    log.Println(bridge.IPAddress)
-
-    GetAllLights(bridge)
-}
-
 type Bridge struct {
     IPAddress   string
     Username    string
diff --git a/bridge_test.go b/bridge_test.go
new file mode 100644
index 0000000..99a169f
--- /dev/null
+++ b/bridge_test.go
@@ -0,0 +1,14 @@
+package hue
+
+// "go test"
+import (
+    "log"
+    "testing"
+)
+
+func TestNewBridge(t *testing.T) {
+    bridge := NewBridge("192.168.1.128", "319b36233bd2328f3e40731b23479207")
+    log.Println(bridge)
+    
+    GetAllLights(bridge)
+}
\ No newline at end of file
diff --git a/lights.go b/lights.go
index 55822fd..1138312 100644
--- a/lights.go
+++ b/lights.go
@@ -1,4 +1,4 @@
-package main
+package hue
 
 import (
     "log"