bug fixes & enhancements