Works in Python3 now