Add docstrings. Make count() a parameter, count.