Fix doc, bugs