Resolved flake8 issues #2345

Task ID: project-989/task-2345
Review ID: 422002
Merge pull request #9 from prakashpp/feature/task-2345

Feature/task 2345