works with python2 and python3 now