Fixed up code to look more pretty and work together a lot more nicely. Added comments where needed, and made main.py actually do something