Add test and benchmark funcs