Adds additional comments and text work.

- Adds additional comments for the models explaining the memoization
  calls.
- PEP8/FLAKE8'ified.