Take into account @agramfort's comments.

Only cosmetic changes.