Adding function to highlight trailing whitespace in vim