First attempt at a main function

I think we have to overthink the for loops in our class methods