Minor updated for python 3