Yeah that's what I need.

I'll do some more testing so my function is similar to the one you tested.