Remove prn statement for count