Work on getting unit_tests to function