Adding python script code