Separate method to initialize token count