Added some docs and also fixed #9