update python script after migrating to ubuntu.