Improved whitespace-mode