Add tests for Util

Test Plan:
- Run tests