Resolved flake8 issues #2343

Review ID: 389001
Merge pull request #14 from priyankarani/feature/task-2343

Resolved flake8 issues #2343