Added testing for exercise 2