fix json import error

Thank you so much for making this! I'm writing a quick script and it came in super handy.

I just fixed the import to work with modern versions of python. ;)