forgot to add the test_images file