Fixed Flake8 issues #4202

Review ID: 2983001
Merge pull request #30 from aroraumang/feature/task-4202

Fixed Flake8 issues #4202