Clean up docs. Fixes #1