Added a few lines in the test, the tests he wrote with a little annotation test all conditions specified