Fixes #1 - don't assume python2