Hello, can you suggest where in the code I can make changes to work on images only instead of videos?