adding unit tests to test data structure