Update doc gen script