Cleanup

Some cleanup to the module and the documentation.