A working main.py