Fixing it up so the count program works