Add reference test code