WIP add generator controller