modified the code to load images from repo