Add CodeGen class