Tweak doc generation script a bit.