Doc generation script update