Fix the ending with new images and the dialogue box