WIP add generator