main.py as class WIP