main.c updated