add bug fixes