Adding unit test code for Lab3