Improve some class documentation