Working on code generator