Docstrings and extra class for DirectReconstruction