Added some code and comment into the test file.