Add code to generate doc