Add a test for the initial import functionality