wip main.py