Add docs to class headers
Closes #660