docs: add more module level documentation + other minor stuff

docs: add more module level documentation + other minor stuff #53