A short cleanup

After reviewing differences between this file and the one in the nilearn repository it seems to my understanding this version is the most up to date and all changes seem to be merged in here.

During the revision I would suggest this minor aesthetic changes.
@GaelVaroquaux, the import is at the top of the file