added mapping for count