modify and finish Q106/main.c