Bug fixes, tweaks