Now compatible with Python3 too