Add file reading code