Add documentation and clean up (#44)