Refactor main function, should now be clearer.

This started due to trying to move to if-let, I quickly realized I
didn't need my continue flag at all.