Add dia support