make it python2/3 compatible