Added additional count function