Added changes to doc gen proccess