Tweaks to jQuery UI init