remove minor changes so code runs in python 2 & 3