Made example python code 2/3 agnostic