Changes to include the file reading function