Improvements to example and docs