Add styles and fix JS error
CSS and JS fixes