updated main.c

updated main.c