Python script change