updated for python 2.x or 3.x

Hi, 
I ran into an issue running this on ubuntu and realised its for python 3 only, I think this should sort your import errors if ran in python 2.x. I tested this and I believe its ok :) hope you don't mind me editing it. If you don't want to use python 2, then no problem

Mark