Add support for stories and an example of use. Fix bug preventing use when installed.

Fix various style issues to comply with PEP 8