Changes in python code