Add equality test support