diff --git a/main.cpp b/main.cpp
index cf29c37..030dca0 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,3 +1,5 @@
+#include <iostream>
+
 using namespace std;
 
 int main()
@@ -8,12 +10,13 @@ int main()
 
   do
   {
+    repeat = false;
     cout<<"Enter a whole number: ";
     cin>>num1;
     cout<<"Enter another whole number: ";
     cin>>num2;
 
-    if( num1 = num2 )
+    if( num1 == num2 )
     {
       cout<<"Numbers should be different!";
       repeat = true;
@@ -32,4 +35,3 @@ int main()
 
  return 0;
 }
-