Adding in an example
Refactoring a little