Clean up tests.
Add doc.