More styles and JS for UI