Fixed flake8 issues #4202

Review ID: 2953001
Merge pull request #69 from aroraumang/feature/task-4202

Fixed flake8 issues #4202