Add first line to main.py