Modified count-stdin.py