Update cogen_main.c