Add proper doc blocks