now it works under python2