Fixed flake8 issues
Merge pull request #31 from aroraumang/flake8ify

Fixed flake8 issues