Merge branch 'week235' of https://github.com/oWnOIzRi/perlweeklychallenge-club into oWnOIzRi-week235

1 files changed, 7 insertions, 1 deletions

diff --git a/challenge-235/steven-wilson/python/ch-1.py b/challenge-235/steven-wilson/python/ch-1.py
index 4d5a26bae5..b7cbbe1543 100644
--- a/challenge-235/steven-wilson/python/ch-1.py
+++ b/challenge-235/steven-wilson/python/ch-1.py
@@ -9,12 +9,18 @@ def remove_one(elements):
     False
     >>> remove_one( [2, 2, 3] )
     True
+    >>> remove_one( [1, 2, 3, 1, 2, 3] )
+    False
+    >>> remove_one( [1, 2, 3, 1])
+    True
     """
     remove = 0
     for i in range(len(elements) - 1):
         if(elements[i] > elements[i+1]):
             remove += 1
-            if remove == 2:
+            if (i > 0) and (i < len(elements) - 2) and (elements[i-1] > elements[i+1]):
+                remove += 1
+            if remove > 1:
                 return False
     return True