bug fixes, tweaks