Update dog_class.py

added an option for user input