make python script better