Modify README generation code