ENH: Speeding up TFCE, cleaning up code