Correct and add documentation