Fixed file locations to work with the code