Added lightbox functionality