Updated lightbox code