Python 3 compatibility

I've updated the imports so that they work with both Python 2 and Python 3 which should address #5