Improve doc blocks and readme