diff --git a/Gopkg.lock b/Gopkg.lock
deleted file mode 100644
index 10ef811..0000000
--- a/Gopkg.lock
+++ /dev/null
@@ -1,9 +0,0 @@
-# This file is autogenerated, do not edit; changes may be undone by the next 'dep ensure'.
-
-
-[solve-meta]
-  analyzer-name = "dep"
-  analyzer-version = 1
-  input-imports = []
-  solver-name = "gps-cdcl"
-  solver-version = 1
diff --git a/Gopkg.toml b/Gopkg.toml
deleted file mode 100644
index 512631a..0000000
--- a/Gopkg.toml
+++ /dev/null
@@ -1,34 +0,0 @@
-# Gopkg.toml example
-#
-# Refer to https://golang.github.io/dep/docs/Gopkg.toml.html
-# for detailed Gopkg.toml documentation.
-#
-# required = ["github.com/user/thing/cmd/thing"]
-# ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"]
-#
-# [[constraint]]
-#   name = "github.com/user/project"
-#   version = "1.0.0"
-#
-# [[constraint]]
-#   name = "github.com/user/project2"
-#   branch = "dev"
-#   source = "github.com/myfork/project2"
-#
-# [[override]]
-#   name = "github.com/x/y"
-#   version = "2.4.0"
-#
-# [prune]
-#   non-go = false
-#   go-tests = true
-#   unused-packages = true
-
-
-[prune]
-  go-tests = true
-  unused-packages = true
-
-[[constraint]]
-  name = "github.com/pkg/errors"
-  version = "0.8.0"
diff --git a/hungarian.go b/hungarian.go
index 369f1a6..6d6dabc 100644
--- a/hungarian.go
+++ b/hungarian.go
@@ -229,6 +229,8 @@ func SolveMax(matrix [][]float64) map[int]map[int]float64 {
 
 	b.setValues()
 
+	b.removeExtra()
+
 	return b.ReducedExtremums
 }
 
diff --git a/hungarian_test.go b/hungarian_test.go
index 4a48d32..4cc8b27 100644
--- a/hungarian_test.go
+++ b/hungarian_test.go
@@ -1,7 +1,7 @@
 package hungarian_test
 
 import (
-	"hungarian"
+	"go-hungarian"
 	"testing"
 )