Make it Python 3 compatible