Add file io for the program