ENH: Add DTI filtering + improve interface