modified wc.py