Added tests to match the quiz