Add documentation to the main class