put more methods in code generator class