Make the repo more modular

[shrn01]

Organise the code into multiple files for better code management. Break up main.py into seperate files based on what each part of code does.