Adding some more structure to your project

I figured instead of sending you an email, I could send you a pull request with some code in it instead