improved the way the code was being tested.