main.py: Add documentation