Fixed flake8 issues #4202

Review ID: 3083001
Merge pull request #1 from aroraumang/feature/task-4202

Fixed flake8 issues #4202