diff --git a/data.pkl b/data.pkl
index 2a3e955..36bfa6b 100644
Binary files a/data.pkl and b/data.pkl differ
diff --git a/main.py b/main.py
index 2ec4a6c..0767942 100644
--- a/main.py
+++ b/main.py
@@ -20,7 +20,7 @@ def train(data):
     D["u"] = D["i"] * D["o"] / D["p"]
     mud_point_idx = D["u"].argmax()
     mud_point = D["lam"].iloc[mud_point_idx]
-    print(f"MUD Point ({mud_point_idx}: {mud_point}")
+    print(f"MUD Point {mud_point_idx}: {mud_point}")
     return mud_point
 
 
@@ -45,5 +45,7 @@ def test(decision=np.array([-0.09, -0.71, -0.43, -0.74]), seed=1992):
 
 if __name__ == "__main__":
     data = pickle.load(open("data.pkl", "rb"))
-    mud_point = train(data)
+    #mud_point = train(data)
+    #mud_point = np.array([ 1, 0, -1, 0])
+    mud_point = np.random.rand(4)*2 - 1
     test(mud_point)