Resolved flake8 issues #2346

Review ID: 392001
Merge pull request #13 from priyankarani/feature/task-2346

Resolved flake8 issues #2346