Small cleanup.
Adding some docs.