Small tweaks on a jQuery plugin