Fix 'spaghetti code'

[jacob-macleod]

Basically, I've written very complex code on 'main.py' with now comments. I'm now trying to debug it, and after taking a week's break, I can't remember what half the variables do. Therefore, at some point, I need to improve the code, and make it actually make sense.