bug fixes
tweaks