Flake8 fixes
Merge pull request #32 from mohanraj1311/flake8fixes

Flake8 fixes