Small changes

Just getting used to the environment, and made a few changes for my own benefit - submitting in case you want to include in the original repo.