Mini bug fixes & enhancements