Updated main with new unit test.