modified py script
modified py script