fix #147

[ABHISHEKSONI121]

Summary

(short description of what does this PR, Issue #, etc.)

Checklist

• [ ] Classes, Variables, function and methods logic ok
• [ ] Comments written explaining what the code does
• [ ] All python code is PEP8 compliant (run black .)
• [ ] No lint issues (run flake8)
• [ ] Test coverage with pytest implemented
• [ ] Reviewers assigned (at least 1 mentor)

Manual test evidence

(attach command-line examples, execution output & logs, etc.)