python code updated