Fixed Flake8 issues #4204

Review Id: 3073001
Merge pull request #29 from goyal1092/feature/task-4204

Fixed Flake8 issues #4204