Make readme generator more flexible