first step in making it python2 compatible