Added a box for image removal