Add unit tests for Homework 1