Add test files back in