update utils
Add test classes