changed python code
swayam fix