Add test for union