Add test for equality