Fixed flake8 issues #4202

Review ID: 3143001
Merge pull request #14 from aroraumang/feature/task-4202

Fixed flake8 issues #4202