Bug fixes/tweaks.