Now only loads first 50 images. This is to handle large datasets.