Add equality check