DOC: clean up the example a bit