Cleaning up a bit of code, added an example level