Add docs/comments