Modified the add_unit_test function, adding a test module.