Fix javascript and add lightbox