added tests for exercise 3