Works with python3 now