Unit Tests for Asgn 2