romove the header file for main
add comments