Bug fixes
Doc update