Added a unit_test() function.