Added unit_test file