edit main.c
small update