Added test code relevant to the current setup.