Add support for Python 3 and 2.6