updated code for python 3 compatibility