Add support for disk type