Certainly! By extending the program's logic, you can count additional digit categories such as prime numbers, multiples of a specific digit, or any other desired classification.