Changed the structure a bit. Now everything is handled in the main.py