Paste the unit test code that I wrote