Small cleanup in docs