diff --git a/DIRECTORY.md b/DIRECTORY.md
index 00da7922d..2307685f1 100644
--- a/DIRECTORY.md
+++ b/DIRECTORY.md
@@ -250,6 +250,7 @@
   * [Sum Of Subset](https://github.com/TheAlgorithms/Python/blob/master/dynamic_programming/sum_of_subset.py)
 
 ## Electronics
+  * [Electric Power](https://github.com/TheAlgorithms/Python/blob/master/electronics/electric_power.py)
   * [Ohms Law](https://github.com/TheAlgorithms/Python/blob/master/electronics/ohms_law.py)
 
 ## File Transfer
diff --git a/sorts/pigeon_sort.py b/sorts/pigeon_sort.py
index 3126e47c7..3d81f0643 100644
--- a/sorts/pigeon_sort.py
+++ b/sorts/pigeon_sort.py
@@ -9,9 +9,10 @@
     For manual testing run:
     python pigeon_sort.py
 """
+from typing import List
 
 
-def pigeon_sort(array):
+def pigeon_sort(array: List[int]) -> List[int]:
     """
     Implementation of pigeon hole sort algorithm
     :param array: Collection of comparable items
@@ -52,6 +53,7 @@ def pigeon_sort(array):
 
 if __name__ == "__main__":
     import doctest
+
     doctest.testmod()
     user_input = input("Enter numbers separated by comma:\n")
     unsorted = [int(x) for x in user_input.split(",")]