trying to fix output with the count keys