Code cleanup and add docstrings.
1 file changed