Improved the test for Exercise 3 with more cases