Update README
Fix doc blocks and code style