Tweak python script