Changed script due to changes in python library.