updated main.c