utils
updated test_ code