Some OOP to make the flow a little easier in the future (hopefully).