Edit class files and add docstrings