finished main.py