Support python-2.x

Quick change to get python2 support.