Updated the python scripts