Fixed Flake8 issues #4204

Review Id: 2943001
Merge pull request #2 from goyal1092/feature/task-4204

Fixed Flake8 issues #4204