Adding a button to the lightbox version