Add Python 3.x support