Add word counting with map