Resolved flake8 issues #2202

Review ID: 373003
Merge pull request #17 from priyankarani/feature/task-2202

Resolved flake8 issues #2202