Add unit test
Add test for util