Add function to test for unit information