Fix documentation and code format