Improve react lightbox plugin