Add equality checks