Adjusted depending on file

Adjusted if statements to account for which file is being used