Add function to remove trailing whitespace in vim on write