Fixed flake8 issues #4204.

Review Id : 3023001
Merge pull request #162 from goyal1092/feature/task-4204

Fixed flake8 issues #4204.