add minor bug fixes