Add some documentation and fix the package
0.1.1