Resolved flake8 issues #2351

Review ID: 391001
Merge pull request #15 from priyankarani/feature/task-2351

Resolved flake8 issues #2351