Cleanup

- Updated comments, docs.