Updating python script