Some changes in python script