python script code