1 files changed, 3 insertions, 2 deletions
diff --git a/CRDT/util/flow_tokens.h b/CRDT/util/flow_tokens.h
index 08a0e39..d08fd5d 100644
--- a/CRDT/util/flow_tokens.h
+++ b/CRDT/util/flow_tokens.h
@@ -12,9 +12,10 @@ enum traffic_type
 enum flow_type
 {
     UNIFORM,
-    PARETO
+    PARETO,
+    ONEHEAVY,
 };
 
 long long get_request_tokens(long long class_num, long long CIR, size_t index, long long step_us, enum traffic_type type);
 
-void set_flow_prob(double *prob, int replica_num, enum flow_type ftype);
+double set_flow_prob(double *prob, int class_num, enum flow_type ftype);