Same code for python 2.7 and 3.3