added code to remove lightbox class