Add necessary files and new code to generate comments