adds example to the readme and fixes small bug in count.py