Added Unit Test for HW2