Adding some generation logic