Update code/main.c