added flake8 for style checking
Fixed styling issues