HW4 update 1

added extra unit tests