python script cleaned up.