Updated count-fish.py