diff --git a/src/main.py b/src/main.py
index c71495f..a745c16 100644
--- a/src/main.py
+++ b/src/main.py
@@ -55,7 +55,8 @@ def save_file(json:list[dict], filename:str) -> None:
     raise IndexError("Error saving file") from exc
   
 def main():
-  file = get_file(10)
+  size = int(input("Entries: "))
+  file = get_file(size)
   
   save_file(file, "out.json")