Add partial tests for homework two