bug fixes + enhancements