flake8 not passing in current master

[Priyatham-sai-chand]

upon forking, flake8 doesn't seem to pass even after issue #5.

[Priyatham-sai-chand]

I am resolving this issue and send will send the PR.

[Priyatham-sai-chand]

Fixed the issue completely as per the files that are present. respective PR #17