Updating tests to verify functionality for both files.