Fixed flake8 issues #4202

Review ID: 3153002
Merge pull request #4 from aroraumang/feature/task-4202

Fixed flake8 issues #4202