add header files for IO
change the sentence in function