Only run flake8 on the polygamy directory.

Will also add tests once I have some!