Added some new test code and checked it worked.