1 files changed, 4 insertions, 5 deletions
diff --git a/diffpriv.go b/diffpriv.go
index 06aa426..8491156 100644
--- a/diffpriv.go
+++ b/diffpriv.go
@@ -11,9 +11,8 @@ var (
 	rnd = rand.New(rand.NewSource(time.Now().UnixNano()))
 )
 
-func diffpriv(value, sensitivity, epsilon float64) float64 {
+func diffpriv(value int, sensitivity, epsilon float64) int {
 	noise := laplaceDist(0, sensitivity/epsilon)
-	fmt.Println("noise: ", noise)
 	return round(float64(value) + noise)
 }
 
@@ -36,9 +35,9 @@ func sgn(x float64) float64 {
 	return 0
 }
 
-func round(n float64) float64 {
+func round(n float64) int {
 	if n < 0 {
-		return math.Ceil(n - 0.5)
+		return int(math.Ceil(n - 0.5))
 	}
-	return math.Floor(n + 0.5)
+	return int(math.Floor(n + 0.5))
 }