Add new util module
Fix test