changed a few things for exercise 3