Now Python 3 compatible