add python2 version. Improve docs slightly