Tidy up lightbox controls