Minor change

variable for count