Adding test for exercise 5 week 6