Modified python files as told by Anil