modify test.c to add main function