PEP8 formatting
Tidy code
Add example