Tweaks, bug fixes