Added the lightbox function