Update code_generator.php