Add a function to get image name