Tests are now PEP8-ified
Add tests to verify count function works