Update main.py

[rowingdude]

Update program structure to a class/dataclass outline and add type hints for intellisense dubugging