Some UI work, CSS fixes, JS improvements