modify improve python