Code cleanup in the docs