Add documentation, export more types