Update source/main.c