Add support for union type syntax