Add DTI models notebook