Update Python generator