Add more styles for buttons, lightbox code