now supports both python2 and python3