Update foundations_hw2_dogs.py