Updating python test script, forgot things.