Make code easier to read by adding comments and docstrings