diff --git a/main.py b/main.py
index fa70098..26955b2 100644
--- a/main.py
+++ b/main.py
@@ -59,10 +59,10 @@ RULES_BOTH = [
     #RE_DOC_COULD_BE_LINK,
 ]
 
-def construct_cmd(rules, target):
+def construct_cmd(rules: list[str], target: str) -> list[str]:
     return ['ag', '--java', '(' + re_select(rules) + ')', target]
 
-if __name__ == '__main__':
+def main():
     args = parser.parse_args()
 
     target = args.project
@@ -73,3 +73,5 @@ if __name__ == '__main__':
     if result1.returncode == 0 or result2.returncode == 0:
         exit(1)
 
+if __name__ == '__main__':
+    main()