main.c testing improvement
main.c testing improvement