Again, did adding extra classes in case of one or two images.