can now return images, fixed single-dataset list prob