Add support for building with Python 3