WIP: better brain mask