adding tests for exercise 4