Slight readability improvements

Added a few comments to make it a little easier to find our way around the code.