Fix comments/documentation