bug fix

**Issue/Feature  Description**
error in the program

**Why this issue to be fixed:**
the expected output not received

**How to reproduce in case of a bug:**
python 

@kirthana-1010 

 

 