Need to fix file reading