diff --git a/main.cpp b/main.cpp
index cf29c37..63f0966 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,33 +1,31 @@
-using namespace std;
+#include <iostream>
 
 int main()
 {
-  int num1;
-  int num2;
-  bool repeat = false;
-
-  do
+  int num1 = 0;
+  int num2 = 0;
+  
+  while(num1 == num2)
   {
-    cout<<"Enter a whole number: ";
-    cin>>num1;
-    cout<<"Enter another whole number: ";
-    cin>>num2;
-
-    if( num1 = num2 )
+    std::cout<<"Enter a whole number: \n";
+    std::cin>>num1;
+    std::cout<<"Enter another whole number: \n";
+    std::cin>>num2;
+    
+    if( num1 == num2 )
     {
-      cout<<"Numbers should be different!";
-      repeat = true;
+      std::cout<<"Numbers should be different!\n";
     }
-  }while(repeat);
+  }
 
-  cout<<"Increasing order: ";
+  std::cout<<"Increasing order: ";
   if( num1 < num2 )
   {
-    cout<<num1<<" "<<num2<<endl;
+    std::cout<<num1<<" "<<num2<<std::endl;
   }
   else
   {
-    cout<<num2<<" "<<num1<<endl;
+    std::cout<<num2<<" "<<num1<<std::endl;
   }
 
  return 0;