updated main.c and main.h