[python] Word count from a txt file program