diff --git a/src/reverselib.py b/src/reverselib.py
index 949f320..04478f7 100644
--- a/src/reverselib.py
+++ b/src/reverselib.py
@@ -7,9 +7,12 @@ def reverse(n):
 
     >>> reverse(234)
     432
+
+    >>> reverse(123)
+    321
     """
     r = 0
-    while n > 1:
+    while n > 0:
         r = r * 10 + n % 10
         n = n // 10
     return r
diff --git a/tests/test_reverseno.py b/tests/test_reverseno.py
index f75b70c..fd20a00 100644
--- a/tests/test_reverseno.py
+++ b/tests/test_reverseno.py
@@ -8,5 +8,13 @@ def test_zero(self):
     def test_ott(self):
         self.assertEqual(reverse(234), 432)
 
+    def test_tva(self):
+        self.assertEqual(reverse(2345), 5432)
+    
+    def test_tre(self):
+        self.assertNotEqual(reverse(234), 234)
+    
+    def test_tre(self):
+        self.assertEqual(reverse(21), 12)
 if __name__ == '__main__':
     unittest.main()