Updating starter code

This is messy. But for time sake, lets leave it.
Regularly, I would have separate files for these different tasks so main would have like 5 lines of code.