correction in python code.