Bug fixes & documentation