Altered new Python script to count