Fixed main.py docstring