modify and finish Q104/main.c