Python2/Python3 compatiable installation

Installation failed on Python3, this fixes that.