Code cleanup #10
Description
There are several minor code quality issues in the latest version of the code. For example, several functions are missing docstrings and there are some whitespace issues throughout the code. We should work to address these issues to improve the code quality.