Committing the changes made to the unit test

Add some comments for the reviewer