Added CodeGenerator class to visitors