diff --git a/src/demo.py b/src/demo.py
index f511597..57567d2 100644
--- a/src/demo.py
+++ b/src/demo.py
@@ -2,8 +2,7 @@ from tpre import *
 
 # 1
 pk_a, sk_a = GenerateKeyPair(1, ())
-m = b'hello world'
-m = int.from_bytes(m)
+m = b"hello world"
 
 # 2
 capsule_ct = Encrypt(pk_a, m)
@@ -11,11 +10,11 @@ capsule_ct = Encrypt(pk_a, m)
 # 3
 pk_b, sk_b = GenerateKeyPair(1, ())
 
-N = 20
-T = 10
+N = 5
+T = 2
 
 # 5
-rekeys =  GenerateReKey(sk_a, pk_b, N, T)
+rekeys = GenerateReKey(sk_a, pk_b, N, T)
 
 # 7
 cfrag_cts = []
@@ -23,9 +22,9 @@ cfrag_cts = []
 for rekey in rekeys:
     cfrag_ct = ReEncrypt(rekey, capsule_ct)
     cfrag_cts.append(cfrag_ct)
-    
+
 # 9
 cfrags = mergecfrag(cfrag_cts)
 m = DecryptFrags(sk_b, pk_b, pk_a, cfrags)
 
-
+print(m)