Slight clean up.
Merge pull request #12 from guardian/clean-up

Slight clean up.