Update main.c

add set test function.