Add doc blocks for classes and methods