1 files changed, 24 insertions, 10 deletions

diff --git a/q8/dh.c b/q8/dh.c
index 074dfc7..1c70c8a 100644
--- a/q8/dh.c
+++ b/q8/dh.c
@@ -2,7 +2,7 @@
 #include <stdlib.h>
 #include <time.h>
 
-// Function to compute a^m mod n
+
 int compute(int a, int m, int n) {
   int r;
   int y = 1;
@@ -12,27 +12,41 @@ int compute(int a, int m, int n) {
     if (r == 1)
       y = (y * a) % n;
     a = a * a % n;
-
     m = m / 2;
   }
 
   return y;
 }
 
-// C program to demonstrate Diffie-Hellman algorithm
+int rand_range(int l, int u) {
+    return rand() % (u - l + 1) + l;
+}
+
+
 int main() {
-  int p = 23; // modulus
-  int g = 5;  // base
-  int a, b; // a - Alice's Secret Key, b - Bob's Secret Key.
-  int A, B; // A - Alice's Public Key, B - Bob's Public Key
+  int p = 23;
+  int g = 5; 
+
+  printf("Modulus: %d, Base: %d\n\n", p, g);
+  int a, b;
+  int A, B;
+
+  int l = 1, u = 1000;
+  printf("RNG range: %d - %d\n\n", l, u);
+
   srand(time(0));
-  a = rand(); // or use rand()
+  a = rand_range(l,u);
   A = compute(g, a, p);
+
   srand(time(0));
-  b = rand(); // or use rand()
+  b = rand_range(l,u);
   B = compute(g, b, p);
+
   int keyA = compute(B, a, p);
   int keyB = compute(A, b, p);
-  printf("\nAlice's Secret Key is %d\nBob's Secret Key is %d\n\n", keyA, keyB);
+
+  printf("Alice's Secret is %d\nBob's Secret is %d\n\n", a, b);
+  printf("Alice's Public Key is %d\nBob's Public Key is %d\n\n", A, B);
+  printf("Alice's Secret Key is %d\nBob's Secret Key is %d\n\n", keyA, keyB);
   return 0;
 }