diff --git a/src/demo.py b/src/demo.py
index 57567d2..f68dc8a 100644
--- a/src/demo.py
+++ b/src/demo.py
@@ -1,17 +1,17 @@
 from tpre import *
 
 # 1
-pk_a, sk_a = GenerateKeyPair(1, ())
+pk_a, sk_a = GenerateKeyPair()
 m = b"hello world"
 
 # 2
 capsule_ct = Encrypt(pk_a, m)
 
 # 3
-pk_b, sk_b = GenerateKeyPair(1, ())
+pk_b, sk_b = GenerateKeyPair()
 
-N = 5
-T = 2
+N = 70
+T = 49
 
 # 5
 rekeys = GenerateReKey(sk_a, pk_b, N, T)