Organize the code

[sebe324]

I have a feeling that the code in this project is really spaghettified. Especially in python. It needs to be fixed

[emipazd]

fixing it, give me a day 🚀