fixup! Add contains function