Updated docstring for main.py