Improving documentation for classes