Added the test method to surround the code.