Add Collatz ref implementation