Fix Cascade dataset to accept both with and without brain mask