Also a test for unlabeled images as the code currently stands?