diff --git a/python/armstrong-numbers/armstrong_numbers.py b/python/armstrong-numbers/armstrong_numbers.py
index 6094c92..cf55ea3 100644
--- a/python/armstrong-numbers/armstrong_numbers.py
+++ b/python/armstrong-numbers/armstrong_numbers.py
@@ -1,2 +1,2 @@
 def is_armstrong_number(number):
-    return sum([int(digit) ** len(str(number)) for digit in str(number)]) == number
+    return sum([int(digit) ** len(str(number)) for digit in str(number)]) == number
\ No newline at end of file
diff --git a/python/atbash-cipher/atbash_cipher.py b/python/atbash-cipher/atbash_cipher.py
index 9a4703e..ffe02a5 100644
--- a/python/atbash-cipher/atbash_cipher.py
+++ b/python/atbash-cipher/atbash_cipher.py
@@ -12,8 +12,8 @@ def groupings(text, groupby=5):
 def sanitize(text):
     return ''.join(text.translate(san).split())
 
-def encode(plain_text):
-    return ' '.join(groupings(sanitize(plain_text.translate(enc).lower())))
+def encode(plain_text, groupby=5):
+    return ' '.join(groupings(sanitize(plain_text.translate(enc).lower()), groupby=groupby))
 
 def decode(ciphered_text):
     return ''.join(ciphered_text.translate(dec).lower().split())
\ No newline at end of file