Adding lightbox support