Added code for new image selection overlay